import pandas as pd

from PD import PdUtil


def get_olt_info():
    df = pd.read_csv("D:\\家宽\\综资\\综资文件(更新至2025年3月4日)\\OLT业务表的配置查询结果.csv", encoding='gbk',
                     usecols=['网元名称', '管理IP', 'OLT归属分公司'])
    return df


# 15分公司处理
def format():
    olt = pd.read_csv(
        "D:\\家宽\\综资\\综资数据(更新至2025年4月29日)\\OLT业务表的配置查询结果.csv", encoding='gbk'
    )
    # olt = pd.read_excel(
    #     "D:\\中断输入\\新建文件夹\\输入文件\\OLT业务表20250217093050.xlsx"
    # )
    olt['分公司'] = olt['所属区县'].str[0:2]
    PdUtil.mask_if_in(olt, '分公司', ['石景', '门头'], '石门')
    PdUtil.mask_if_in(olt, '分公司', ['东城', '西城'], '中区')
    olt.loc[(olt['管理单位'] == '经开分公司') & (olt['分公司'].isin(['大兴', '通州'])), '分公司'] = '经开'
    olt.to_csv('D:\\中断输入\\OLT业务表的配置查询结果.csv', index=False)

    olt = olt[['网管侧网元名称', '网元状态', '管理IP', '分公司']]
    olt.to_excel("D:\\中断输入\\新建文件夹\\OLT业务表.xlsx", index=False)
    print(olt)


if __name__ == '__main__':
    format()
